const fs = require('fs');
const path = require('path');

/**
 * 获取当前目录下所有的 MP4 文件的绝对路径
 * @returns {string[]} MP4文件的绝对路径列表
 */
function getMp4List() {
    const currentDir = process.cwd();
    const files = fs.readdirSync(currentDir);
    
    return files
        .filter(file => path.extname(file).toLowerCase() === '.mp4')
        .map(file => path.resolve(currentDir, file));
}

/**
 * 创建 videos.txt 文件，写入所有 MP4 文件的路径
 * 格式为: file '文件路径'
 * 如果文件已存在会被覆盖
 */
function writeVideoList() {
    const videos = getMp4List();
    const content = videos
        .map(filepath => `file '${filepath}'`)
        .join('\n');
    fs.writeFileSync('videos.txt', content, 'utf8');
}

module.exports = {
    getMp4List,
    writeVideoList
}; 